import datetime
import os.path
import sys

from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from webdriver_manager.chrome import ChromeDriverManager


class Driver:
    driver=""
    def __init__(self):
        options = webdriver.ChromeOptions()
        options.page_load_strategy = "normal"
        self.driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()),options=options)

    def getScreenShot(self):
        #截图命名方式：调用的方法名-2024-3-28-150233
        #不同日期的截图放在不同目录
        dirname = datetime.datetime.now().strftime("%Y-%m-%d")
        #如果目录不存在就创建
        if not os.path.exists("../images/" + dirname):
            os.mkdir("../images/" + dirname)
        #图片名：调用的方法名-2024-3-28-150233
        filename = sys._getframe().f_back.f_code.co_name + "-" + datetime.datetime.now().strftime("%Y-%m-%d-%H%M%S") + ".png"
        #截图
        self.driver.save_screenshot(f'../images/{dirname}/' + filename)
ChatDriver=Driver()